python train_one_for_all_attr_fix_arm.py --epochs 800 \
                                         --lr 1e-6 \
                                         --gpu 1 \
                                         --data arm-pool2d.pkl \
                                         --label_path tvm_data/arm_data/pool2d_data/labels.txt \
                                         --op_type MaxPooling2D \
                                         --attr_name pool_size \
                                         --attr_index 0 \
                                         --model v3 \
                                         --log_level INFO \
                                         --eval_per_epoch 2 \
                                         --ckpt /home/kxp/workspace/tvm-reversion/results/2022-03-20-14-37-53-\(train_one_for_all_attr_fix_arm.py\)/checkpoints/model_ckpt_399.pth
